#ifndef UCM_CNCLMEM_H_
#define UCM_CNCLMEM_H_

#include "cnrt.h"

cnrtRet_t ucm_cnrtMalloc(void **pPtr, size_t bytes);
cnrtRet_t ucm_orig_cnrtMalloc(void **pPtr, size_t bytes);
cnrtRet_t ucm_override_cnrtMalloc(void **pPtr, size_t bytes);

cnrtRet_t ucm_cnrtHostMalloc(void **pPtr, size_t bytes);
cnrtRet_t ucm_orig_cnrtHostMalloc(void **pPtr, size_t bytes);
cnrtRet_t ucm_override_cnrtHostMalloc(void **pPtr, size_t bytes);

cnrtRet_t ucm_cnrtFree(void *ptr);
cnrtRet_t ucm_orig_cnrtFree(void *ptr);
cnrtRet_t ucm_override_cnrtFree(void *ptr);

cnrtRet_t ucm_cnrtFreeHost(void *ptr);
cnrtRet_t ucm_orig_cnrtFreeHost(void *ptr);
cnrtRet_t ucm_override_cnrtFreeHost(void *ptr);

#endif
